home *** CD-ROM | disk | FTP | other *** search
/ PC World 2006 February / PCWorld_2006-02_cd.bin / software / vyzkuste / triky / triky.exe / httrack-3.33.exe / {app} / src / htsnostatic.h < prev    next >
C/C++ Source or Header  |  2005-02-05  |  8KB  |  279 lines
  1. /* ------------------------------------------------------------ */
  2. /*
  3. HTTrack Website Copier, Offline Browser for Windows and Unix
  4. Copyright (C) Xavier Roche and other contributors
  5.  
  6. This program is free software; you can redistribute it and/or
  7. modify it under the terms of the GNU General Public License
  8. as published by the Free Software Foundation; either version 2
  9. of the License, or any later version.
  10.  
  11. This program is distributed in the hope that it will be useful,
  12. but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. GNU General Public License for more details.
  15.  
  16. You should have received a copy of the GNU General Public License
  17. along with this program; if not, write to the Free Software
  18. Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19.  
  20.  
  21. Important notes:
  22.  
  23. - We hereby ask people using this source NOT to use it in purpose of grabbing
  24. emails addresses, or collecting any other private information on persons.
  25. This would disgrace our work, and spoil the many hours we spent on it.
  26.  
  27.  
  28. Please visit our Website: http://www.httrack.com
  29. */
  30.  
  31.  
  32. /* ------------------------------------------------------------ */
  33. /* File: htsnostatic.c subroutines:                             */
  34. /*       thread-safe routines for reentrancy                    */
  35. /* Author: Xavier Roche                                         */
  36. /* ------------------------------------------------------------ */
  37.  
  38. /*
  39.   Okay, with these routines, the engine should be fully reentrant (thread-safe)
  40.   All static references have been changed:
  41.  
  42.   from
  43.   function foo() {
  44.     static bartype bar;
  45.   }
  46.   to:
  47.   function foo() {
  48.     bartype* bar;
  49.     NOSTATIC_RESERVE(bar, bartype, 1);
  50.   }
  51. */
  52.  
  53. #ifndef HTSNOSTATIC_DEFH
  54. #define HTSNOSTATIC_DEFH 
  55.  
  56. /* Library internal definictions */
  57. #ifdef HTS_INTERNAL_BYTECODE
  58.  
  59. #include "htscore.h"
  60. #include "htsthread.h"
  61.  
  62. #define HTS_VAR_MAIN_HASH 127
  63.  
  64. /*
  65.   MutEx
  66. */
  67.  
  68.  
  69. /* Magic per-thread variables functions 
  70.  
  71.   Example:
  72.   hts_lockvar();
  73.   hts_setvar("MyFoo", (long int)(void*)&foo);
  74.   hts_unlockvar();
  75.   ..
  76.   foo=(void*)(long int)hts_directgetvar("MyFoo");
  77.  
  78.   Do not forget to initialize (hts_initvar()) the library once per thread
  79. */
  80. int hts_initvar(void);
  81. int hts_freevar(void);
  82. #ifndef HTTRACK_DEFLIB
  83. HTSEXT_API int hts_resetvar(void);
  84. #endif
  85. int hts_maylockvar(void);
  86. int hts_lockvar(void);
  87. int hts_unlockvar(void);
  88.  
  89. int hts_setvar(char* name, long int value);
  90. int hts_getvar(char* name, long int* ptrvalue);
  91. long int hts_directgetvar(char* name);
  92.  
  93. int hts_setblkvar(char* name, void* value);
  94. int hts_getblkvar(char* name, void** ptrvalue);
  95. void* hts_directgetblkvar(char* name);
  96.  
  97. /* Internal */
  98. int hts_setextvar(char* name, long int value, int flag);
  99. int hts_getextvar(char* name, long int* ptrvalue, int flag);
  100. void hts_destroyvar(void* ptrkey);
  101. void hts_destroyvar_key(void* adr);
  102.  
  103. /* 
  104.   Ensure that the variable 'name' has 'nelts' of type 'type' reserved 
  105.   fnc is an UNIQUE function name
  106. */
  107. #define NOSTATIC_RESERVE(name, type, nelt) NOSTATIC_XRESERVE(name, type, nelt)
  108.  
  109. /*
  110.   Note:
  111.   Yes, we first read the localInit flag variable without MutEx protection,
  112.   for optimization purpose, because the flag is set once initialization DONE.
  113.   If the first read fails, we *securely* re-check and initialize *if* necessary.
  114.   The abort() things should NEVER be called, and are here for safety reasons
  115. */
  116. /*
  117.   function-specific static cKey:
  118.   cKey = { localKey, localInit }
  119.               ||        \
  120.               \/          \ ==1 upon initialization
  121.          thread variable
  122.               ||
  123.               \/
  124.              void*
  125.               ||
  126.               \/
  127.       'thread-static' value
  128.  
  129.   the function-specific static cKey is also referenced in the global 
  130.   hashtable for free() purpose: (see hts_destroyvar())
  131.  
  132.       global static key variable
  133.         'hts_static_key'
  134.                ||
  135.                \/
  136.          thread variable
  137.                ||
  138.                \/
  139.               void*
  140.                ||
  141.                \/
  142.             hashtable
  143.                ||
  144.                \/
  145.      function-specific hash key
  146.                ||
  147.                \/
  148.               &cKey
  149.  
  150. */
  151. #ifdef _WIN32
  152.  
  153. #ifdef _WIN32_WCE
  154.  
  155. /* Windows CE: static only */
  156. #define NOSTATIC_XRESERVE(name, type, nelt) do { \
  157.   /*__declspec( thread )*/ static type thValue[nelt]; \
  158.   /* __declspec( thread ) */ int static initValue = 0; \
  159.   name = thValue; \
  160.   if (!initValue) { \
  161.     initValue = 1; \
  162.     memset(&thValue, 0, sizeof(thValue)); \
  163.   } \
  164. } while(0)
  165.  
  166. #elif 1
  167.  
  168. /* New Windows version: TLS */
  169. /* Suggested by daan at zwif.com to be more gentle with LoadLibrary (04/2004)
  170. See http://msdn.microsoft.com/library/en-us/vccore/html/_core_rules_and_limitations_for_tls.asp
  171. And especially the "DLL declares any nonlocal data or object as __declspec( thread )" section
  172. */
  173. #define NOSTATIC_XRESERVE(name,type,nelt) do { \
  174.   static DWORD tlsIndex = 0; \
  175.   static   int initValue = 0; \
  176.   if (initValue == 0) \
  177.   { \
  178.     if (!hts_maylockvar()) { \
  179.       abortLog("unable to lock mutex (not initialized?!)"); \
  180.       abort(); \
  181.     } \
  182.     hts_lockvar(); \
  183.     if (initValue == 0) { \
  184.       tlsIndex = TlsAlloc(); \
  185.       if (tlsIndex == 0xFFFFFFFF) { \
  186.         abortLog("unable to allocate thread local storage (TLS) for variable!"); \
  187.         abort(); \
  188.       } \
  189.       initValue = 1; \
  190.     } \
  191.     hts_unlockvar(); \
  192.   } \
  193.   name = (type*)TlsGetValue(tlsIndex); \
  194.   if (name == NULL) { \
  195.     name = (type*)malloc(sizeof(type)*nelt); \
  196.     if (name == NULL) { \
  197.       abortLog("unable to allocate memory for variable!"); \
  198.       abort(); \
  199.     } \
  200.     memset(name, 0, sizeof(type)*nelt); \
  201.     TlsSetValue(tlsIndex, name); \
  202.   } \
  203. } while(0)
  204.  
  205. #else
  206.  
  207. /* Windows: handled by the compiler */
  208. #define NOSTATIC_XRESERVE(name, type, nelt) do { \
  209.   __declspec( thread ) static type thValue[nelt]; \
  210.   __declspec( thread ) int  static initValue = 0; \
  211.   name = thValue; \
  212.   if (!initValue) { \
  213.     initValue = 1; \
  214.     memset(&thValue, 0, sizeof(thValue)); \
  215.   } \
  216. } while(0)
  217.  
  218. #endif
  219.  
  220. #else
  221.  
  222. /* Un*x : slightly more complex, we have to create a thread-key */
  223. typedef struct {
  224.   PTHREAD_KEY_TYPE localKey;
  225.   unsigned char localInit;
  226. } hts_NostaticComplexKey;
  227. #define NOSTATIC_XRESERVE(name, type, nelt) do { \
  228. static hts_NostaticComplexKey cKey={0,0}; \
  229. name = NULL; \
  230. if ( cKey.localInit ) { \
  231.   PTHREAD_KEY_GET(cKey.localKey, &name, type*); \
  232. } \
  233. if ( ( ! cKey.localInit ) || ( name == NULL ) ) { \
  234.   if (!hts_maylockvar()) { \
  235.     abortLog("unable to lock mutex (not initialized?!)"); \
  236.     abort(); \
  237.   } \
  238.   hts_lockvar(); \
  239.   { \
  240.     { \
  241.       name = (type *) calloc((nelt), sizeof(type)); \
  242.       if (name == NULL) { \
  243.         abortLog("unable to allocate memory for variable!"); \
  244.         abort(); \
  245.       } \
  246.       { \
  247.         char elt_name[64+8]; \
  248.         sprintf(elt_name, #name "_%d", (int) __LINE__); \
  249.         PTHREAD_KEY_CREATE(&(cKey.localKey), NULL); \
  250.         hts_setblkvar(elt_name, &cKey); \
  251.       } \
  252.       PTHREAD_KEY_SET(cKey.localKey, name, type*); \
  253.       name = NULL; \
  254.       PTHREAD_KEY_GET(cKey.localKey, &name, type*); \
  255.       if (name == NULL) { \
  256.         abortLog("unable to load thread key!"); \
  257.         abort(); \
  258.       } \
  259.       if ( ! cKey.localInit ) { \
  260.         cKey.localInit = 1; \
  261.       } \
  262.     } \
  263.   } \
  264.   hts_unlockvar(); \
  265. } \
  266. else { \
  267.   PTHREAD_KEY_GET(cKey.localKey, &name, type*); \
  268.   if (name == NULL) { \
  269.     abortLog("unable to load thread key! (2)"); \
  270.     abort(); \
  271.   } \
  272. } \
  273. } while(0)
  274. #endif
  275.  
  276. #endif
  277.  
  278. #endif
  279.